from pydantic import BaseModel


class ModelInput(BaseModel):
    api_key: str
    user_input: str
    history_id: int

class GeneratorInput(BaseModel):
    api_key: str
    subject: str
    time: int
    temperature: float

class xhs_GeneratorInput(BaseModel):
    api_key: str
    subject: str
    target: str

